# docker build -t test-sts:sft .

FROM nvcr.io/nvidia/pytorch:24.01-py3
COPY LLaMA-Factory.tar.gz /workspace/
WORKDIR /workspace
RUN tar zxf LLaMA-Factory.tar.gz && rm -f LLaMA-Factory.tar.gz && \
  cd LLaMA-Factory && \
  pip install --no-cache-dir -e .[metrics] -i https://pypi.tuna.tsinghua.edu.cn/simple && \
  pip install flash-attn --no-build-isolation
CMD ["/bin/sleep", "infinity"]

